Spiral geometry using python